package com.yhx.log;

import org.apache.log4j.Logger;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import java.util.Enumeration;

@Aspect        //定义一个切面
@Component  //配置Bean
public class WebLogAspect {
    private Logger logger = Logger.getLogger(getClass());
    
    private long startTime=0;
    //定义切入点为controllor下的所有的方法
    @Pointcut("execution(public * com.yhx.controller.*.*(..))")
    //com.myspringboot.controller  为Controller所在的包名
    public void webLog() {

    }

    @Before("webLog()")
    public void doBefore(JoinPoint joinPoint) throws Throwable {
        //哪个方法(方法的全限定名，包含返回值和参数)
        Signature signature = joinPoint.getSignature();
        startTime = System.currentTimeMillis();
        // 接收到请求，记录请求内容
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = attributes.getRequest();
        // 记录下请求内容
        logger.info("---------------request----------------");
        logger.info("方法名："+signature);
        logger.info("URL : " + request.getRequestURL().toString());
        logger.info("HTTP_METHOD : " + request.getMethod());
        logger.info("IP : " + request.getRemoteAddr());
        Enumeration enu = request.getParameterNames();
        while (enu.hasMoreElements()) {
            String name = (String) enu.nextElement();
            logger.info("name:" + name + " - value:" + request.getParameter(name));
        }
    }


    @AfterReturning(returning = "ret", pointcut = "webLog()")
    public void doAfterReturning(Object ret) throws Throwable {
        logger.info("---------------response----------------");
        // 处理完请求，返回内容
        logger.info("RESPONSE : " + ret);
        logger.info("花费时间 : " + (System.currentTimeMillis() - startTime) + "毫秒");
    }
    
    // 通过@Pointcut定义切入点，此处是com.test.controller包下的所有Controller（对controller层所有请求处理做切入点），
    // 然后通过@Before实现对请求内容的日志记录，最后通过@AfterReturning记录请求返回的对象。
    // 这样所有功能的请求参数，返回数据，以及运行时间等都可以统一记录，不需要在每个方法去单独处理。
}
